package com.ideaplugin.datamanager.controller;

import com.ideaplugin.datamanager.model.common.PlainResult;
import org.dromara.x.file.storage.core.FileInfo;
import org.dromara.x.file.storage.core.FileStorageService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;

@RestController
@RequestMapping("/file")
public class FileDetailController {

    @Autowired
    private FileStorageService fileStorageService;//注入实列

    /**
     * 上传文件
     */
    @PostMapping("/upload")
    public PlainResult<FileInfo> upload(MultipartFile file) {
        return new PlainResult<>(fileStorageService.of(file).upload());
    }

    /**
     * 上传文件，成功返回文件 url
     */
    @PostMapping("/upload2")
    public PlainResult<String> upload2(MultipartFile file) {
        FileInfo fileInfo = fileStorageService.of(file)
                .setPath("upload/") //保存到相对路径下，为了方便管理，不需要可以不写
                .setObjectId("0")   //关联对象id，为了方便管理，不需要可以不写
                .setObjectType("0") //关联对象类型，为了方便管理，不需要可以不写
                .putAttr("role", "admin") //保存一些属性，可以在切面、保存上传记录、自定义存储平台等地方获取使用，不需要可以不写
                .upload();  //将文件上传到对应地方
        return new PlainResult<>(fileInfo == null ? "上传失败！" : fileInfo.getUrl());
    }

    /**
     * 上传图片，成功返回文件信息
     * 图片处理使用的是 https://github.com/coobird/thumbnailator
     */
    @PostMapping("/uploadImage")
    public PlainResult<FileInfo> uploadImage(MultipartFile file) {
        final FileInfo upload = fileStorageService.of(file)
                .image(img -> img.size(1000, 1000))  //将图片大小调整到 1000*1000
                .thumbnail(th -> th.size(200, 200))  //再生成一张 200*200 的缩略图
                .upload();
        return new PlainResult<>(upload);
    }

    /**
     * 上传文件到指定存储平台，成功返回文件信息
     */
    @PostMapping("/upload-platform")
    public PlainResult<FileInfo> uploadPlatform(MultipartFile file) {
        final FileInfo upload = fileStorageService.of(file)
                .setPlatform("aliyun-oss-1")    //使用指定的存储平台
                .upload();
        return new PlainResult<>(upload);
    }

    /**
     * 直接读取 HttpServletRequest 中的文件进行上传，成功返回文件信息
     * 使用这种方式有些注意事项，请查看文档 基础功能-上传 章节
     */
    @PostMapping("/upload-request")
    public PlainResult<FileInfo> uploadPlatform(HttpServletRequest request) {
        return new PlainResult<>(fileStorageService.of(request).upload());
    }

}
